package _interview100;

import _object.ListNode;

/**
 * 23. 合并 K 个升序链表
 */
public class No23 {
    /**
     * 1. 朴素（顺序）
     */
    public ListNode mergeKLists1(ListNode[] lists) {
        ListNode ans = null;

        for (ListNode list : lists) {
            ans = mergeTwoLists(ans, list);
        }
        return ans;
    }

    /**
     * 2. 分治（归并）
     */
    public ListNode mergeKLists(ListNode[] lists) {
        return mergeKLists(lists, 0, lists.length - 1);
    }

    /**
     * 根据中点拆分链表
     */
    private ListNode mergeKLists(ListNode[] lists, int l, int r) {
        if (l > r) return null; // 注意输入的 lists 可能是空的
        if (l == r) return lists[l]; // 无需合并，直接返回
        int mid = l + (r - l) / 2;
        ListNode left = mergeKLists(lists, l, mid); // 合并左半部分
        ListNode right = mergeKLists(lists, mid + 1, r); // 合并右半部分
        return mergeTwoLists(left, right); // 最后把左半和右半合并
    }

    /**
     * 合并两个有序链表
     */
    private ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        ListNode dummy = new ListNode(-1), cur = dummy;
        while (list1 != null && list2 != null) {
            if (list1.val <= list2.val) {
                cur.next = list1;
                list1 = list1.next;
            } else {
                cur.next = list2;
                list2 = list2.next;
            }
            cur = cur.next;
        }
        cur.next = list1 != null ? list1 : list2;
        return dummy.next;
    }
}
